/*
 * Copyright (c) 2020 - present, Inspur Genersoft Co., Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.inspur.edp.bff.designtime.generator.core.rtgen.datagenerator.propertygenerator;

import com.inspur.edp.caf.generator.baseInfo.TypeInfo;
import com.inspur.edp.caf.generator.property.ClassPropertyGenerator;
import com.inspur.edp.cef.entity.entity.EntityDataCollection;
import com.inspur.edp.cef.gencommon.CefPropertyInfo;
import org.eclipse.jdt.core.dom.*;

import java.util.ArrayList;

public class ChildCollectionPropGenerator extends ClassPropertyGenerator {
    private String privatePropName;
    private CefPropertyInfo propertyInfo;

    public ChildCollectionPropGenerator(String privatePropName, CefPropertyInfo propertyInfo) {
        this.privatePropName = privatePropName;
        this.propertyInfo = propertyInfo;
    }

    @Override
    protected ArrayList<Modifier.ModifierKeyword> getGetterAccessModifier() {
        return new java.util.ArrayList<Modifier.ModifierKeyword>(java.util.Arrays.asList(new Modifier.ModifierKeyword[]{Modifier.ModifierKeyword.PUBLIC_KEYWORD}));
    }

    @Override
    protected ArrayList<Modifier.ModifierKeyword> getSetterAccessModifier() {
        return new java.util.ArrayList<Modifier.ModifierKeyword>(java.util.Arrays.asList(new Modifier.ModifierKeyword[]{Modifier.ModifierKeyword.PUBLIC_KEYWORD}));
    }

    @Override
    protected String getPropertyName() {
        return propertyInfo.getPropertyName();
    }

    @Override
    protected TypeInfo getPropertyType() {
        return new TypeInfo(propertyInfo.getPropertyMethod().getReturnType());
    }

    @Override
    protected String getFieldName() {
        return privatePropName;
    }

    @Override
    protected boolean hasGetMethod() {
        return true;
    }

    @Override
    protected boolean hasSetMethod() {
        return false;
    }

    @Override
    protected Block buildGetMethodBody() {
        Block block = ast.newBlock();
        buildNewIfNull(block);
        ReturnStatement returnStatement = ast.newReturnStatement();
        returnStatement.setExpression(ast.newSimpleName(privatePropName));
        block.statements().add(returnStatement);
        return block;
    }

    private void buildNewIfNull(Block block) {
        IfStatement ifStatement = ast.newIfStatement();
        buildIfExpression(ifStatement);
        buildIfThen(ifStatement);
        block.statements().add(ifStatement);
    }

    private void buildIfExpression(IfStatement ifStatement) {
        InfixExpression infixExpression = ast.newInfixExpression();
        infixExpression.setLeftOperand(ast.newSimpleName(privatePropName));
        infixExpression.setOperator(InfixExpression.Operator.EQUALS);
        infixExpression.setRightOperand(ast.newNullLiteral());
        ifStatement.setExpression(infixExpression);
    }

    private void buildIfThen(IfStatement ifStatement) {
        Assignment assignment = ast.newAssignment();
        assignment.setLeftHandSide(ast.newSimpleName(privatePropName));
        assignment.setOperator(Assignment.Operator.ASSIGN);

        ClassInstanceCreation classInstanceCreation = ast.newClassInstanceCreation();
        classInstanceCreation.setType(ast.newSimpleType(ast.newName(EntityDataCollection.class.getTypeName())));
        assignment.setRightHandSide(classInstanceCreation);
        ifStatement.setThenStatement(ast.newExpressionStatement(assignment));
    }
}
